import os

from application.vk_base_mcp_v2.yuque.request import YuqueBaseRequest

doc_urls_dict = {
    'introduction': ['axr72hit2947cigh'],
    'entity.go': ['/pn6fx4i5az9vkp43', '/vbgsq64zes5210b5'],
    'value.go': ['/wyxtoxfvrbsnykoz'],
    'parser.go': ['/xag7bdrndfrbqlge'],
    'validate.go': ['/tg4qqkgayznzecry'],
    'adapter.go': ['/rsgxtxz3rwhsquyf'],
}

yuque_request = YuqueBaseRequest('shuofeng09', 'pub4o1', doc_urls_dict)


class VKCodingPrompt:

    @staticmethod
    def entity_go_prompt() -> str:
        return yuque_request.fetch_content_by_file_name('entity.go')

    @staticmethod
    def value_go_prompt():
        return yuque_request.fetch_content_by_file_name('value.go')

    @staticmethod
    def parser_go_prompt():
        return yuque_request.fetch_content_by_file_name('parser.go')

    @staticmethod
    def validate_go_prompt():
        return yuque_request.fetch_content_by_file_name('validate.go')

    @staticmethod
    def adapter_go_prompt():
        return yuque_request.fetch_content_by_file_name('adapter.go')
